package com.secxiot.common.aspect;

import cn.hutool.json.JSONUtil;
import com.secxiot.common.utils.IdWorker;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.slf4j.MDC;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;

@Aspect
@Component
public class WebLogAspect {

    private static final Logger logger = LoggerFactory.getLogger(WebLogAspect.class);

    @Autowired
    private IdWorker idWorker;

    @Pointcut("execution( * com.secxiot..controller.*.*(..))")//两个..代表所有子目录，最后括号里的两个..代表所有参数
    public void logPointCut() {
    }


    @Before("logPointCut()")
    public void doBefore(JoinPoint joinPoint) throws Throwable {
        MDC.put("thread", idWorker.nextId().toString());
        // 接收到请求，记录请求内容
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();

        // 记录下请求内容
        logger.info("请求地址 : " + request.getRequestURL().toString());
        logger.info("HTTP METHOD : " + request.getMethod());
        // 获取真实的ip地址
        //logger.info("IP : " + IPAddressUtil.getClientIpAddress(request));
        logger.info("CLASS_METHOD : " + joinPoint.getSignature().getDeclaringTypeName() + "."
                + joinPoint.getSignature().getName());
        Object[] joinPointArgs = joinPoint.getArgs();
        String args = JSONUtil.toJsonStr(joinPointArgs);
        if (args.length() > 1000) {
            StringBuilder sb = new StringBuilder("[");
            for (int i = 0; i < joinPointArgs.length; i++) {
                String str = JSONUtil.toJsonStr(joinPointArgs[i]);
                if (str.length() > 200) {
                    sb.append("\n").append(str.substring(0, 200)).append("...., \n");
                } else {
                    sb.append(str).append(", ");
                }
            }
            sb.append("]");
            logger.info("参数 : " + sb.toString());
        } else {
            logger.info("参数 : " + JSONUtil.toJsonStr(args));
        }
//        loggger.info("参数 : " + joinPoint.getArgs());

    }

    @AfterReturning(returning = "ret", pointcut = "logPointCut()")// returning的值和doAfterReturning的参数名一致
    public void doAfterReturning(Object ret) throws Throwable {
        // 处理完请求，返回内容(返回值太复杂时，打印的是物理存储空间的地址)
        logger.debug("返回值 : " + JSONUtil.toJsonStr(ret));
    }

    @Around("logPointCut()")
    public Object doAround(ProceedingJoinPoint pjp) throws Throwable {
        long startTime = System.currentTimeMillis();
        Object ob = pjp.proceed();// ob 为方法的返回值
        logger.info("耗时 : " + (System.currentTimeMillis() - startTime));
        MDC.remove("thread");
        return ob;
    }
}
